# session_info数据类，用户保存每一轮对话的数据
from dataclasses import dataclass
from typing import List, Dict


@dataclass
class SessionInfo:
    session_id: str
    case_id: int
    customer_id: str
    workspace_id: int
    dialogue_id: int
    faq_id: int
    company_id: int
    customer_base_info: Dict[str, str]
    collect_info: Dict[str, int]
    lastest_psychology_state: str
    lastest_user_target: str
    formulate_solution: str
    lastest_process_stage: str
    context: List[Dict[str, str]]
    extends_info: Dict[str, int]
